package com.huihe.framework.service;

import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.huihe.framework.vo.SysRoleVo;
import com.huihe.framework.mapstruct.SysRoleMapStruct;
import com.huihe.storage.entity.SysRole;
import com.huihe.storage.entity.SysRoleMenu;
import com.huihe.storage.repository.SysRoleMenuRepository;
import com.huihe.storage.repository.SysRoleRepository;
import lombok.AllArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;

/**
 * 角色 Service 层
 */
@Slf4j
@Service
@AllArgsConstructor
public class SysRoleService {

    /**
     * 角色 映射
     */
    SysRoleMapStruct sysRoleMapStruct;

    /**
     * 角色 Repository 层
     */
    SysRoleRepository sysRoleRepository;

    /**
     * 角色和菜单关联 Repository 层
     */
    SysRoleMenuRepository sysRoleMenuRepository;

    /**
     * 根据 ID 查询
     * @return 角色
     */
    public SysRoleVo findById(String id){
        SysRole sysRole = sysRoleRepository.getById(id);
        return sysRoleMapStruct.entityToVo(sysRole);
    }

    /**
     * 条件查询-不分页
     * @return 查询结果
     */
    public List<SysRoleVo> findByList(Map<String, Object> map){
        List<SysRole> list = sysRoleRepository.findByList(map);
        return sysRoleMapStruct.entityListToVoList(list);
    }

    /**
     * 条件查询-分页
     * @return 查询结果
     */
    public IPage<SysRoleVo> findByPage(Map<String, Object> map){
        IPage<SysRole> page = sysRoleRepository.findByPage(map);
        return sysRoleMapStruct.entityPageToVoPage(page);
    }

    /**
     * 创建
     * @return 创建数据数量
     */
    public boolean create(SysRoleVo vo){
        SysRole entity = sysRoleMapStruct.voToEntity(vo);
        sysRoleRepository.saveOrUpdate(entity);
        //存入角色关联菜单
        List<SysRoleMenu> roleMenus = new ArrayList<>();
        if (StrUtil.isNotBlank(entity.getId())){
            for (String menuId : vo.getMenuIds()){
                SysRoleMenu roleMenu = new SysRoleMenu();
                roleMenu.setRoleId(entity.getId());
                roleMenu.setMenuId(menuId);
                roleMenus.add(roleMenu);
            }
            sysRoleMenuRepository.saveBatch(roleMenus);
            return true;
        }
        return false;
    }

    /**
     * 更新
     * @return 更新数据数量
     */
    public boolean update(SysRoleVo vo){
        SysRole entity = sysRoleMapStruct.voToEntity(vo);
        sysRoleRepository.updateById(entity);
        //存入角色关联菜单
        List<SysRoleMenu> roleMenus = new ArrayList<>();
        if (StrUtil.isNotBlank(entity.getId())){
            for (String menuId : vo.getMenuIds()){
                SysRoleMenu roleMenu = new SysRoleMenu();
                roleMenu.setRoleId(entity.getId());
                roleMenu.setMenuId(menuId);
                roleMenus.add(roleMenu);
            }
            sysRoleMenuRepository.saveBatch(roleMenus);
            return true;
        }
        return false;
    }

    /**
     * 根据ID删除
     * @return 删除数据数量
     */
    public boolean deleteById(String id){
        return sysRoleRepository.removeById(id);
    }
}
